package com.tanhua.dubbo.util;

import com.tanhua.domain.mongo.Sequence;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.mongodb.core.FindAndModifyOptions;
import org.springframework.data.mongodb.core.MongoTemplate;
import org.springframework.data.mongodb.core.query.Criteria;
import org.springframework.data.mongodb.core.query.Query;
import org.springframework.data.mongodb.core.query.Update;
import org.springframework.stereotype.Component;

/**
 * id自增器，是用于发布动态时，给动态的pid字段赋值。
 * 这个字段是给大数据系统使用的，要求所有动态数据的pic全局唯一自增。
 * 所以提供一个ID自增器
 * @author anshuai
 * @create 2021-01-30 15:20
 */
@Component
public class IdSequeue {

    @Autowired
    private MongoTemplate mongoTemplate;

    /**
     * 获取指定集合的自增序列值
     * @param collName 集合名称
     * @return 自增的序列值
     */
    public Long getNextId(String collName){
        //创建查询条件
        Query query = new Query(Criteria.where("collName").is(collName));
        Update update = new Update();
        //mongodb中$inc是用于字段更新操作上的
        //更新字段seqId,每次自增1
        update.inc("seqId", 1);

        //设置选项
        FindAndModifyOptions options = new FindAndModifyOptions();
        //  如果集合里没有此记录，是否新增进去
        options.upsert(true);
        //  是否修改后获取最新的值
        options.returnNew(true);

        Sequence sequence = mongoTemplate.findAndModify(query, update, options, Sequence.class);

        return sequence.getSeqId();
    }

}
